#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@File     : params.py
@Project  : pipecoco
@Date     : 2021/8/13
@Author   : Zhang Jinyang
@Contact  : zhang-jy@sjtu.edu.cn
'''

layers_config = {
    'alexnet': ['conv1', 'relu', 'max_pool2d', 'conv2', 'relu', 'max_pool2d','conv3', 'relu','pad', 'conv4', 'relu', 'conv5', 'relu', 'max_pool2d','flatten', 'fc1', 'relu', 'fc2', 'relu', 'fc3'],
    'vgg_small': ['layers', 'flatten', 'classifier'],
    'vgg_large': ['layers', 'flatten', 'classifier']
}

input_shapes = {
    'vgg_small': [3, 32, 32],
    'vgg_large': [3, 128, 128],
    'alexnet': [3, 227, 227],
    'resnet34': [3, 224, 224]
}

divided_points = {
    "resnet34": 2,
    "vgg_small": 3,
    "vgg_large": 3,
    "alexnet": 2
}

